package com.zr.rabbitmq.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * RabbitMQ消息确认配置
 * 包括生产者确认和消息返回
 */
@Slf4j
@Component
public class RabbitMQConfirmConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 初始化方法，设置确认回调和返回回调
     */
    @PostConstruct
    public void init() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 消息发送到交换机确认回调
     *
     * @param correlationData 相关配置信息
     * @param ack             交换机是否成功收到消息
     * @param cause           失败原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String messageId = correlationData != null ? correlationData.getId() : "未知ID";
        
        if (ack) {
            log.info("消息发送到交换机成功，消息ID: {}", messageId);
        } else {
            log.error("消息发送到交换机失败，消息ID: {}, 失败原因: {}", messageId, cause);
            // 这里可以实现消息重发或持久化到数据库等补偿机制
        }
    }

    /**
     * 消息从交换机发送到队列失败时的回调
     * 只有当消息无法路由到队列时才会触发
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error(
                "消息从交换机发送到队列失败: 消息内容: {}, 响应码: {}, 响应信息: {}, 交换机: {}, 路由键: {}",
                new String(message.getBody()), replyCode, replyText, exchange, routingKey
        );
        // 这里可以实现消息重发或持久化到数据库等补偿机制
    }
}
